package com.wchat.config;

import org.springframework.boot.web.servlet.MultipartConfigFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.util.unit.DataSize;
import org.springframework.util.unit.DataUnit;

import javax.servlet.MultipartConfigElement;

/**
 * @author : willian fu
 * @date : 2021/9/2
 */
@Configuration
public class FileUploadConfig {

    @Bean
    public MultipartConfigElement multipartConfigElement() {
        MultipartConfigFactory factory = new MultipartConfigFactory();
        // 单个数据大小
        factory.setMaxFileSize(DataSize.of(100L, DataUnit.MEGABYTES)); // KB,MB
        // 总上传数据大小
        factory.setMaxRequestSize(DataSize.of(100L, DataUnit.MEGABYTES));
        return factory.createMultipartConfig();
    }
}
